import sys
import traceback
from xjlibrary.all_step.BaseAllStep import GlobalControl
from xjlibrary.our_file_dir import BaseDir

curPath = BaseDir.get_file_dir_absolute(__file__)
print(curPath)
sys.path.insert(0, curPath)  # 插入
from Step1_save_t import main as step1_main
from Step2_GenYearPage import main as step2_main
from Step3_save_dir import main as step3_main
from Step4_save_id import main as step4_main
from Step5_page_detail_down import main as step5_main

"""
需要可以访问cnki的代理 一般使用xxnet
"""


class UseGlobalControl(GlobalControl):

    def __init__(self):
        super().__init__(__file__, BaseDir.get_new_path(BaseDir.get_file_dir_absolute(__file__), "logs"))

    def imports(self):
        # from Step1_save_t import main as step1_main
        # from Step2_GenYearPage import main as step2_main
        # from Step3_save_dir import main as step3_main
        # from Step4_save_id import main as step4_main
        # from Step5_page_detail_down import main as step5_main
        pass

    def run(self, strings):
        """
        该步骤运行代码
        :return:
        """
        try:
            if strings == "":
                return True
            result = eval(strings)
        except:
            print(traceback.format_exc())
            return False
        return result

    def set_step1_start(self):
        """
        将第一步的结果归0
        :return:
        """
        sql = "update `journal` set stat=0 WHERE stat=1"
        self.mysqlutils.ExeSqlToDB(sql)

    def start_hook(self, key):
        if key == "5":
            self.key = "5"
            self.nowstep = ""
            return True

    def one_check(self):
        """
        该步骤检查代码是否运行完成
        :return:
        """
        if self.key == "1":
            print("检查第一步")
            sql = "SELECT subid FROM subclass"
            rows = self.mysqlutils.SelectFromDB(sql)
            from Step1_save_t import HtmlSubClass, HtmlYearRoot
            HtmlSubClassnum = BaseDir.get_dir_file_num(HtmlSubClass)
            HtmlYearRootnum = BaseDir.get_dir_file_num(HtmlYearRoot)
            if len(rows) == HtmlYearRootnum and HtmlSubClassnum == 10:
                print("执行正常")
                self.writetrue()
            else:
                self.writefalse("量不对")
        elif self.key == "2":
            print("检查第二步")
            self.writetrue("没有好的验证机制 默认通过")
        elif self.key == "3":
            self.writetrue()
        elif self.key == "4":
            sql = "select stat from issue where stat=0"
            rows = self.mysqlutils.SelectFromDB(sql)
            if not rows:
                self.writetrue()
            else:
                self.writefalse("有issue没有下载 请检查原因")
        elif self.key == "5":
            self.writetrue()

        if "5" in self.jsondicts.keys() and self.jsondicts["5"]["isFinish"]:
            return True
        return False


if __name__ == "__main__":
    globalcontrol = UseGlobalControl()
    globalcontrol.work()
